Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Refactor fine-tuning #112

Merged
merged 10 commits into from
Oct 8, 2024
Merged

Refactor fine-tuning #112

merged 10 commits into from
Oct 8, 2024

Conversation

mattwoodx
Copy link
Contributor

  • Fine-tuning is now a standalone class which has multiple inheritance
  • This allows the fine-tuning classes to access parent process_data and get_embeddings methods
  • Doesn't require instantiation of a parent class explicitly, we rather pass configs to the fine-tuning model

@mattwoodx mattwoodx marked this pull request as draft October 7, 2024 14:12
@mattwoodx mattwoodx added the enhancement New feature or request label Oct 7, 2024
@mattwoodx mattwoodx marked this pull request as ready for review October 7, 2024 15:20
@@ -13,14 +13,14 @@

logger = logging.getLogger(__name__)

class GeneformerFineTuningModel(HelicalBaseFineTuningModel):
class GeneformerFineTuningModel(HelicalBaseFineTuningModel, Geneformer):
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Very nice!

@bputzeys bputzeys assigned bputzeys and unassigned bputzeys Oct 8, 2024
@bputzeys bputzeys merged commit fe843a2 into main Oct 8, 2024
2 checks passed
@bputzeys bputzeys deleted the refactor-fine-tuning branch October 8, 2024 09:29
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants